package yaq.mini.weibo.leafwb.util;

import com.auth0.jwt.JWT;
import com.auth0.jwt.JWTCreator;
import com.auth0.jwt.algorithms.Algorithm;
import com.auth0.jwt.interfaces.DecodedJWT;

import java.util.Calendar;
import java.util.Map;

/**
 * @author HMQ
 * @date 2020/9/20 13:32
 */
public class JwTUtil {

    private static final String SIGN = "$!WUSS&*@SS&^45";

    /**
     * 获取token，获得的token会在7天后过期
     *
     * @param map 需要存入的payload信息
     * @return token令牌
     */
    public static String getToken(Map<String, String> map) {
        Calendar calendar = Calendar.getInstance();
        //设置7天后的时候
        calendar.add(Calendar.DATE, 7);

        //创建jwt builder
        JWTCreator.Builder builder = JWT.create();

        //创建payload
        map.forEach((k, v) -> {
            builder.withClaim(k, v);
        });

        //设置7天后过期
        String token = builder.withExpiresAt(calendar.getTime())
                .sign(Algorithm.HMAC256(SIGN));

        return token;
    }

    /**
     * 验证token令牌的合法性，如果令牌错误或者过期就会返回异常
     *
     * @param token 要验证的token令牌
     * @return 令牌合法则返回DecodedJWT，可获取token信息，不合法则产生异常
     */
    public static DecodedJWT vertify(String token) {
        return JWT.require(Algorithm.HMAC256(SIGN)).build().verify(token);
    }
}
